#!/usr/bin/env bash

# Train and test STAViS audiovisual models for all splits

base_path='./experiments/audiovisual_train_test'

results_path='results'
mkdir -p ${base_path}'/'${results_path}


python main_train_one_dataset.py --gpu_devices 0,1 --batch_size 128 --n_threads 12 \
    --use_transformer \
    --num_heads 4 \
    --result_path results_nlevels3_fuxian \
    --nlevels 3 \
    --audiovisual True --checkpoint 20 --n_epochs 60 \
    --root_path ${base_path} \
    --pretrain_path ./data/pretrained_models/resnet-50-kinetics.pth  \
    --audio_pretrain_path ./data/pretrained_models/soundnet8.pth \
    --annotation_path_av_train ./data/fold_lists/AVD_list_train_fps.txt \
    --annotation_path_av_test ./data/fold_lists/AVD_list_test_fps.txt


# python main_train_one_dataset.py --gpu_devices 0,1 --batch_size 128 --n_threads 12 \
#     --use_transformer \
#     --num_heads 4 \
#     --result_path results_nlevels5 \
#     --nlevels 5 \
#     --audiovisual True --checkpoint 20 --n_epochs 60 \
#     --root_path ${base_path} \
#     --pretrain_path ./data/pretrained_models/resnet-50-kinetics.pth  \
#     --audio_pretrain_path ./data/pretrained_models/soundnet8.pth \
# 	  --annotation_path_av_train ./data/fold_lists/AVD_list_train_fps.txt \
#     --annotation_path_av_test ./data/fold_lists/AVD_list_test_fps.txt

# python main_train_one_dataset.py --gpu_devices 0,1 --batch_size 128 --n_threads 12 \
#     --use_transformer \
#     --num_heads 4 \
#     --result_path results_nlevels3 \
#     --nlevels 3
#     --audiovisual True --checkpoint 20 --n_epochs 60 \
#     --root_path ${base_path} \
#     --pretrain_path ./data/pretrained_models/resnet-50-kinetics.pth  \
#     --audio_pretrain_path ./data/pretrained_models/soundnet8.pth \
# 	  --annotation_path_av_train ./data/fold_lists/AVD_list_train_fps.txt \
#     --annotation_path_av_test ./data/fold_lists/AVD_list_test_fps.txt
#
# python main_train_one_dataset.py --gpu_devices 0,1 --batch_size 128 --n_threads 12 \
#     --use_transformer \
#     --num_heads 4 \
#     --result_path results_nlevels7 \
#     --nlevels 7 \
#     --audiovisual True --checkpoint 20 --n_epochs 60 \
#     --root_path ${base_path} \
#     --pretrain_path ./data/pretrained_models/resnet-50-kinetics.pth  \
#     --audio_pretrain_path ./data/pretrained_models/soundnet8.pth \
# 	  --annotation_path_av_train ./data/fold_lists/AVD_list_train_fps.txt \
#     --annotation_path_av_test ./data/fold_lists/AVD_list_test_fps.txt
